from PIL import Image, ImageDraw
import os

def create_simple_icon():
    """创建一个简单的图标文件"""
    # 确保resources目录存在
    os.makedirs("resources", exist_ok=True)
    
    # 图标文件路径
    ico_path = "resources/icon.ico"
    
    # 如果图标已存在，则跳过
    if os.path.exists(ico_path):
        print(f"图标文件已存在: {ico_path}")
        return True
    
    try:
        # 创建一个256x256的图像
        img = Image.new('RGBA', (256, 256), color=(255, 255, 255, 0))
        draw = ImageDraw.Draw(img)
        
        # 绘制一个简单的图形（例如，一个彩色方块）
        draw.rectangle([(50, 50), (206, 206)], fill=(0, 120, 215))
        draw.rectangle([(70, 70), (186, 186)], fill=(255, 255, 255))
        draw.rectangle([(90, 90), (166, 166)], fill=(0, 120, 215))
        
        # 保存为ICO文件
        img.save(ico_path, format='ICO')
        
        print(f"成功创建图标文件: {ico_path}")
        return True
    except Exception as e:
        print(f"创建图标时出错: {str(e)}")
        return False

if __name__ == "__main__":
    create_simple_icon()